# -*- coding: utf-8 -*-

import warnings

import pandas as pd
from sklearn import preprocessing

from ..base import CommonFunction



class FeatureExtraction(CommonFunction):
    def linear_features(self):
        # todo
        pass

    def polynomial_features(self, degree=2, interaction_only=False, include_bias=True):
        poly = preprocessing.PolynomialFeatures(degree=degree, interaction_only=interaction_only, include_bias=include_bias)
        self._data = pd.DataFrame(poly.fit_transform(self._data))
        return poly
